#pragma once
#include <functional>
#include <iostream>
#include <cstdint>
#include <cstring>
#include <fmt/core.h>
#include <pthread.h>
#include <string>
#include <unistd.h>

using namespace fmt;

namespace ThreadModlue
{
    static uint32_t number = 1;
    class Thread
    {
        // 定义一个函数类型，用于存储线程函数
        using func_t = std::function<void()>;

    private:
        /**
         * @brief 启用线程分离功能
         *
         * 该函数用于将线程设置为分离状态，并输出一条消息表示线程已被分离。
         * 分离状态的线程在结束时其资源会被自动回收。
         */
        void EnableDetach()
        {
            println("线程被分离了"); // 输出消息，表示线程已被分离
            _isdetach = true;        // 将线程状态设置为分离
        }
        /**
         * @brief 启用运行状态
         *
         * 该函数用于将对象的运行状态设置为启用。调用此函数后，
         * 对象的_isrunning成员变量将被设置为true，表示对象处于运行状态。
         */
        void EnableRunning()
        {
            _isrunning = true;
        }
        /**
         * @brief 线程执行例程函数
         *
         * 该函数是线程启动后执行的例程函数，负责线程的初始化、分离以及执行用户定义的函数。
         *
         * @param args 传入的参数，通常是一个指向Thread对象的指针
         * @return 返回值为nullptr，表示线程执行结束
         */
        static void *Routine(void *args)
        {
            // 将传入的参数转换为Thread指针
            Thread *self = static_cast<Thread *>(args);

            // 启用线程运行状态
            self->EnableRunning();

            // 如果线程设置为分离状态，则进行分离操作
            if (self->_isdetach)
                self->Detach();

            // 执行用户定义的函数
            self->_func();

            // 返回nullptr，表示线程执行结束
            return nullptr;
        }

    public:
        /**
         * @brief 构造函数，用于初始化线程对象。
         *
         * @param func 线程要执行的可调用对象。
         */
        Thread(func_t func) : _tid(0), _isdetach(false), _isrunning(false), _func(func)
        {
            /**
             * @brief 初始化线程名称。
             *
             * 线程名称由 "thread-" 前缀和自增的编号组成。
             * 例如，第一个线程的名称为 "thread-0"，第二个线程的名称为 "thread-1"，依此类推。
             */
            _name = "thread-" + std::to_string(number++);
        }
        // 分离线程
        /**
         * @brief 分离当前线程
         *
         * 该函数用于分离当前线程，使其在完成后自动释放资源。
         * 如果线程已经处于分离状态，则函数直接返回。
         * 如果线程正在运行，则调用pthread_detach函数进行分离。
         * 最后，调用EnableDetach函数来更新线程状态。
         */
        void Detach()
        {
            // 检查线程是否已经处于分离状态
            if (_isdetach)
            {
                return;
            }
            // 检查线程是否正在运行
            if (_isrunning)
                // 调用pthread_detach函数分离线程
                pthread_detach(_tid);
            // 更新线程状态为分离状态
            EnableDetach();
        }
        // 启动线程
        /**
         * @brief 启动线程。
         *
         * 该函数用于启动一个新的线程来执行指定的任务。如果线程已经处于运行状态，
         * 则返回 false。否则，尝试创建一个新的线程，并返回创建结果。
         *
         * @return 如果线程成功创建并启动，则返回 true；否则返回 false。
         */
        bool Start()
        {
            // 检查线程是否已经在运行
            if (_isrunning)
                return false;

            // 创建新线程，执行 Routine 函数，并将 this 传递给它
            int n = pthread_create(&_tid, nullptr, Routine, this);

            // 检查线程创建是否成功
            if (n != 0)
            {
                // 如果创建失败，输出错误信息
                std::cerr << "creat thread error: " << strerror(n) << std::endl;
                return false;
            }
            else
            {
                // 如果创建成功，输出成功信息
                println("{} start success", _name);
                return true;
            }
        }
        /**
         * @brief 停止正在运行的线程。
         *
         * 该函数用于停止一个正在运行的线程。首先检查线程是否正在运行，
         * 如果是，则尝试取消线程。如果取消线程成功，则将线程状态设置为停止，
         * 并输出停止信息；如果取消线程失败，则输出错误信息并返回失败状态。
         *
         * @return 如果线程成功停止，返回 true；否则返回 false。
         */
        bool Stop()
        {
            if (_isrunning)
            {
                // 尝试取消线程
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    // 如果取消线程失败，输出错误信息
                    std::cerr << "create thread error: " << strerror(n) << std::endl;
                    return false;
                }
                else
                {
                    // 如果取消线程成功，更新线程状态并输出停止信息
                    _isrunning = false;
                    println("{} stop", _name);
                    return true;
                }
            }
            // 如果线程未运行，直接返回 false
            return false;
        }
        /**
         * @brief 尝试将当前线程与指定的线程进行连接（join）。
         *
         * 该函数首先检查目标线程是否已经被分离（detached）。如果已经被分离，则输出错误信息并返回。
         * 否则，调用pthread_join函数尝试将当前线程与目标线程进行连接。如果连接失败，输出错误信息；
         * 如果连接成功，输出成功信息。
         */
        void Join()
        {
            // 检查线程是否已经被分离
            if (_isdetach)
            {
                // 输出错误信息：线程已经被分离，不能进行join
                println("你的线程已经是分离的了,不能进行join");
                return;
            }
            // 调用pthread_join函数尝试进行线程连接
            int n = pthread_join(_tid, &res);
            // 检查pthread_join的返回值
            if (n != 0)
                // 如果返回值不为0，表示连接失败，输出错误信息
                std::cerr << "create thread error: " << strerror(n) << std::endl;
            else
                // 如果返回值为0，表示连接成功，输出成功信息
                println("join success");
        }
        ~Thread()
        {
        }

    private:
        pthread_t _tid;    //  线程ID
        std::string _name; //  线程名称
        bool _isdetach;    //  是否分离
        bool _isrunning;   //  是否运行
        void *res;         //  返回值
        func_t _func;      //  线程函数
    };
} // namespace ThreadModlue
